import os
from dotenv import load_dotenv

load_dotenv()

# 模型配置
QWEN_API_KEY = os.getenv("QWEN_API_KEY")
QWEN_API_BASE = os.getenv("QWEN_API_BASE")

# Milvus配置
MILVUS_HOST = "192.168.1.20"
MILVUS_PORT = 19530
COLLECTION_NAME = "pdf_documents"

# 向量维度
TEXT_EMBEDDING_DIM = 512
IMAGE_EMBEDDING_DIM = 512

# PDF处理配置
PDF_UPLOAD_DIR = "uploads" 